Refactor model's class info retrieval