Got add model functionality working