added test.py
saving work