Custom rake task to add unit_class to db