ENH: updated script