Added figure output