PEP8 fixes on the figure module

Here is a small patch for pep8 fixes on the figure module.

Thanks,
N