Improve the shorts_controller.rb file