Small update to comments