Add wake function