python create class in function