Updated the data cleaning script