main.rb - added maths