Cleaning up main module