Correction to preparation.rb week4