Fixed graph size and graph output

Fixed graph size and graph output