Updated Code Comments

Updated the comments in the code to reflect the more recent changes