Improvements on code documentation