Add trace plots to SIR inference example