Reboot handler cleanup

Nitpicking on the timeout style, but I think this is more ruby:
```
require 'timeout'
begin
  Timeout.timeout(20) do
    loop do
      ...
    end
  end
rescue Timeout::Error
  STDERR.puts "Node state never reached valid state. Exit"
  exit(1)
end
```