Fixed code documentation