change inheritence structure to live in main.rb