Add missing clean_data_dir function