Use get_class() to work with custom models

To work with custom models need to load them at runtime