Updated the source tree for to use with tensor data