Improve init.sh