Some small code changes 
Picture for the flow_chart