add cleanup call