Clean up layout.rb