adding location to the model