addition to the code

calculator