Rail Fence Code In Python