Minor update to comments