Add remove_pyc function