Add browser wait back in