Add rails4 support