Small change to flow chart