Added a function for out-of-sample embedding