added mock functionality to perform inference