Added models for each controller