Update display.rb