Clean up code comment