added code to calculator.rb