Intro to python homework-simple programs