New Models and Code Update

Added new models, and updated small parts of code