renamed problem1 python