add/update unit tests