Rewrite for rails 4.0