Add enum to the highlighted activerecord functions