Updates to TritonUnit