Updated QueryBuilder to extend ModelCompanionGenerator instead of the removed AbstractModelGenerator