Create a calculator in python