worked on adding graph features