Added argument hook for clean function