Cleaning up code comments