[IMP]Improved code as per pep8 coding style.