Simple Python Math Program