Moving matplotlib import inside function and docstring