Add simple signal handling