Updated figure script