Add rails 4 support