Made set_env.sh more automatic