Changed model to be more rails4