Added in update tests and fixed a bug.