Updated NNManager with corrected input parameter count