Add support for escape key