Adding the block diagram