added models and repositoy class