Add TP specific wrapper

Remove extra whitespace after `function`