added more models and fixed some code