Added code for loading and cleaning the data