Add some cleanup function.