Add mock location support