Validated rest according to PEP8
Unified importing of pyplot to "plt"