add await and defer to highlighting