[IMP]Improved code as per Pep8.