Cleanup comments

Some cleanup that I forgot