Add models code