Clean main and rename problem5.rb