Hold wake lock while shutting down.

It is nicer to make sure the screen stays on while we are shutting
down, so the screen goes off when we are actually complete rather
than some time before.

Change-Id: I8725ac9884df6d21344f35288da3e375d9779b3b