clean up tensor module and complete tests