Fixed model code and added a new method.