Python Assignment Solution on Object Oriented