Cleanup plots 1,3,4 code