add 3 extra unit tests