Add wakelock mechanism