Make things prettier
Measure inference time