Add type documentation