Edited model class

Updated the model class to incorporate adding, and retrieving models
from the dictionary