Document more code