working on the inference portion