Fixed some snytax errors and added more functionality to the main.rb