Update script - add tests.