Fixed the end error for main.rb