Set up classes for assignment