diff --git a/pixel.cpp b/pixel.cpp
index ac6c0bd2ebda14bbc31dc284aa56b7b90b8305f6..247f1ca3728d27fcd647883e2d60573e0fac078c 100644
--- a/pixel.cpp
+++ b/pixel.cpp
@@ -9,24 +9,29 @@ typedef ap_axis<32,0,0,0> pkt_t;
 
 
 void pixel(
-
+		ap_int<32> w,
 		hls::stream< pkt_t > &din,
 		hls::stream< pkt_t > &dout
-
 ) {
 	#pragma HLS INTERFACE ap_ctrl_none port=return
+	#pragma HLS INTERFACE s_axilite port=w
 	#pragma HLS INTERFACE axis port=din
 	#pragma HLS INTERFACE axis port=dout
 
+	int i=0;
 	pkt_t pkt;
 	din.read(pkt);
-    if(pkt.data == 1){
-		pkt.data = 215;
+
+	if(pkt.data){
+
+		i=i+1;
+	}
+
+	if(i==w){
+		pkt.data*=2;
 	}
-	if(pkt.data == 215){
-		pkt.data = 1;
-	}       
 
 	dout.write(pkt);
 }
 
+