Add ability to ctrl-C