Updated Comments

Updated documentation to cover new code