Added desired functionality.

https://gist.github.com/8r2y5/5ea9023278e17c25e0c6 - pylint