Add Graph functionality