Adds support for rails 4