Add handler for SIGTERM
Fix tests