Add test for lib