Add drag function for the map