Removed a lot of code which is not necessary when only working with data set one