Clean up KO scripts