Square And Cube In Python