more changes to include tensortype