how to write a class in python