[IMP] : code as per pep8