Add the option to extend the base model and collection classes