Allow stopping test.py with ctrl-C