adds comments to and updates main.rb and shelter.rb