Update grasstest.py