python class definition in function