Updated block diagram