Refactor 001.rb