Improve deploy script