Cleaned up code documentation