returning values from function class 12 python