diff --git a/models.go b/models.go
deleted file mode 100644
index 06ab7d0..0000000
--- a/models.go
+++ /dev/null
@@ -1 +0,0 @@
-package main
diff --git a/models/models.go b/models/models.go
index 842a6eb..0402400 100644
--- a/models/models.go
+++ b/models/models.go
@@ -13,30 +13,30 @@ type Answer struct {
 	// resp time?
 }
 
-type OpenRouterResp struct {
-	ID       string `json:"id"`
-	Provider string `json:"provider"`
-	Model    string `json:"model"`
-	Object   string `json:"object"`
-	Created  int    `json:"created"`
-	Choices  []struct {
-		Logprobs           any    `json:"logprobs"`
-		FinishReason       string `json:"finish_reason"`
-		NativeFinishReason string `json:"native_finish_reason"`
-		Index              int    `json:"index"`
-		Message            struct {
-			Role      string `json:"role"`
-			Content   string `json:"content"`
-			Refusal   any    `json:"refusal"`
-			Reasoning any    `json:"reasoning"`
-		} `json:"message"`
-	} `json:"choices"`
-	Usage struct {
-		PromptTokens     int `json:"prompt_tokens"`
-		CompletionTokens int `json:"completion_tokens"`
-		TotalTokens      int `json:"total_tokens"`
-	} `json:"usage"`
-}
+// type OpenRouterResp struct {
+// 	ID       string `json:"id"`
+// 	Provider string `json:"provider"`
+// 	Model    string `json:"model"`
+// 	Object   string `json:"object"`
+// 	Created  int    `json:"created"`
+// 	Choices  []struct {
+// 		Logprobs           any    `json:"logprobs"`
+// 		FinishReason       string `json:"finish_reason"`
+// 		NativeFinishReason string `json:"native_finish_reason"`
+// 		Index              int    `json:"index"`
+// 		Message            struct {
+// 			Role      string `json:"role"`
+// 			Content   string `json:"content"`
+// 			Refusal   any    `json:"refusal"`
+// 			Reasoning any    `json:"reasoning"`
+// 		} `json:"message"`
+// 	} `json:"choices"`
+// 	Usage struct {
+// 		PromptTokens     int `json:"prompt_tokens"`
+// 		CompletionTokens int `json:"completion_tokens"`
+// 		TotalTokens      int `json:"total_tokens"`
+// 	} `json:"usage"`
+// }
 
 type DSResp struct {
 	ID      string `json:"id"`
diff --git a/parser.go b/parser.go
index 127dab7..84e6700 100644
--- a/parser.go
+++ b/parser.go
@@ -54,6 +54,7 @@ func (p *deepSeekParser) MakePayload(prompt string) io.Reader {
 	  "stream_options": null,
 	  "suffix": null,
 	  "temperature": 1,
+      "n_probs": 10,
 	  "top_p": 1
 	}`, prompt))
 }