Add more efficient trace norm