Updated code comments for better doc generation