simple python program to calculate numbers