Update main.cpp

Removed graph image code. Part of issue #90 work.