Add signal handling for SIGINT