added my assignment 2 model