Commenting some code, added better reboot method