Changed graph generation class from inheritance to composite