diff --git a/src/detection/mod.rs b/src/detection/mod.rs
index 78a857b..cc9a15a 100644
--- a/src/detection/mod.rs
+++ b/src/detection/mod.rs
@@ -1,3 +1,5 @@
+//! Detection module, consisting of segformer implementation
+
 mod segformer;
 
 use crate::error::Result;
diff --git a/src/detection/segformer.rs b/src/detection/segformer.rs
index b91e598..45e6fc3 100644
--- a/src/detection/segformer.rs
+++ b/src/detection/segformer.rs
@@ -1,3 +1,4 @@
+//! Segformer implementation
 use candle_core::{Module, ModuleT, Result, Tensor, D};
 use candle_nn::{
     conv2d, conv2d_no_bias, layer_norm, linear, Activation, Conv2d, Conv2dConfig, Linear,
diff --git a/src/recognition/mod.rs b/src/recognition/mod.rs
index 090a510..f28b3f9 100644
--- a/src/recognition/mod.rs
+++ b/src/recognition/mod.rs
@@ -16,7 +16,6 @@ use swin_transformer::SwinModel;
 pub struct Config {
     encoder: SwinConfig,
     decoder: MBartConfig,
-    pad_token_id: i32,
 }
 
 #[derive(Debug, Clone)]