[demo] Modified cleanup script