dryed code up by using regular models