Add escape functionality