Modified 'Adding the class' section