Updated the model not found exception class
Fixed a typo
Fixed another typo
Complete the generator refactoring