calc.py - enter problem and do math with functions