add code for calculator