adding support for rails 4