Added models for user story 1 MVP