Update script, fixes #1