Update stress.py

giving stress.py a main function