please could you already adopt the new plt style. Directly import after numpy