Little refactor to preparation.rb