changed test_class.py example to use args for function calls