Shift Array Python Practice