Improved comments

Some improvements in code comments.