Update

Accidentally included some test code.