Add support for generating graphs from output