Added signal handler. Kinda.