started adding in model classes