add more models and make associations work