Refactored the code to migrate to Rails 4.0.1.