How to write code for simple calculator