Added partial wake-lock support