fix_graph

#92 
@marwa-mehanna  @hishamelgezeery 
please review