added classes to main.rb