Update

Accidentally included test code.