add reboot too

Change-Id: I55af8870ba9b02ee311e23ab63456ffcec5bc444