Create class_test.py