change main for env_assist.py

Change-Id: I430ae69890f00a9bbc335d1b2ff323e093ec3f51