Improve mark.sh