Add check for argument to dmg script