Updated cleanData function