DrawGraph class works, still, needs refactoring