Add auto cleanup