Add test for embedding