Added unitID to User model