Add signal handler
fix