Cleanup after module 2 completion