Added model parameter obj
Finished part 2