update tests and stuff