Fixed errors in UML diagram

Cleaned up main.rb