Adding the ability to use rails 4