Adding support for rails 4