Added more code documentation