Array Shifting Code Python