added data cleaning