Added output graph
Deleted unused model classes