Clean up code documentation