Fix Defining classes part 2