Moving test.py script to tests/ directory