Unit tests add