An Easy Problem Python