updated comments for code