Tried to fix the graphs