Add type in TODO example