Add trace plots to SI inference example