Fix comments as suggested by @ElBrogrammer