Added basic inference code incomplete