add data cleaning code