Add support for rails 4