Updated reference to main script