Add class to get current location