Made changes to data cleaning