Add immediate shutdown feature