Update script to accept version