Fixed units and added max enrollment
main.py units