add extra unit tests