Refactor codebase - Change all enrollment dicts to Enrollment objects, remove unused imports