adding python class