Updated the notes and added comments