Improve production.rb