Can you add a `@cleanup` decorator to this function, please?