Adding location to the model