basic maths in python.