update tests

Good idea. Updating