fixed type bug
Improved Embedding