Add clean up code.