Added inference engine class