Updated comments

Improved comments in code