Example - code from assignment 8 - multiple classes