Changed way of printing PNG for plots 3 and 4

Fixed truncated legend in plots 3 and 4

Also used setup to check if data was already loaded to speed up plots
after the first one.