update/add tests