New pull request for PR#85

Hi.

I have merged the changes with the main repo and github says that this can be merged with the main repo automatically. I have also followed PEP8 style now everywhere in the code