Refactoring Procedural Code into Object Oriented Code in Python