update main.sh