Got graceful shutdown working :)