Removed graph display
Moved generic functions to util.py
Removed unnecessary display graph