Cleanup of code documentation