Add minor adjustments to milestone script