Fixed typo and PEP8
Optional import of plotting functions