Thinkful python class assignment