Continued work on project 3. Updated to work for more than 2 classes.