Python Class Method To Solve