Update test.py

Add shebang