fixing clean-up for python files