cleaned up code then remade plots