diff --git a/src/human.rs b/src/human.rs
index f289ad7..d4dab2f 100644
--- a/src/human.rs
+++ b/src/human.rs
@@ -1,7 +1,7 @@
 // use crate::prelude::*;
 
 // #[derive(Copy, Clone, PartialEq)]
-#[derive(PartialEq)]
+#[derive(PartialEq, Debug)]
 pub enum State {
     Normal,
     Infected,
diff --git a/src/population.rs b/src/population.rs
index 8cf8d8d..70a3394 100644
--- a/src/population.rs
+++ b/src/population.rs
@@ -21,6 +21,16 @@ pub fn human_idx(x: i32, y: i32, width: i32) -> usize {
     ((y * width) + x)as usize
 }
 
+#[test]
+fn popuplation_gen() {
+    let disease = Disease::new(20,10,5,String::from("Covid 44"));
+    let population = Population::new(20,10,5,5,7,disease);
+    assert_eq!(population.humans.len(), 5 * 7);
+    for human in population.humans.iter() {
+        assert_eq!(human.present_state, State::Normal);
+    }
+}
+
 impl Population{
     pub fn new(start_infected_ratio:u32,start_immune_ratio:u32,start_dead_ratio:u32,width:i32,height:i32,plague:Disease)->Self{
         let mut the_humans: Vec<Human> = Vec::with_capacity((width*height) as usize);