First Python Program Input Definition Math