Slightly corrected diagram output