Add functionality to calculator