add models code