Add signal_handler to setup