simplifying plot 2 code and improving png