including test.py file
modified last example.