Slight modification to compile script