cleaned up the code and fixed the model. added "Assignment".