Add more code documentation