Added model and mock data