add models in homework 4