update comments and Problem2.rb is working and tested