Added models for appropriate migrations