ValueError when trying to write a for loop in python