[IMP]Improved code as per pep8 standard