Move matplotlib imports inside function
Ensure that ImageNormalize is importable (but not necessarily functional) even if matplotlib is not installed.