Improve script for creating diskimage