Improve publish.sh