updated models for sprint 2