Modify script for custume build