Add data clean up