add wake up functionality