Fixed screen problem on both types of graph outputs