added main class to store graph