Comments update

Updated comments in code