Took a hack at some inference stuff