Correction to preparation.rb week 4