Add support for Rails4