Python Edabit Problem