Minor cleaning up in comments