problem 20 was easy in python