Implement proper input prediction class