Class 6 - added notes within the code