Loading Data and Training the model