Add polling feature