Clean up main