diff --git a/README.md b/README.md
index f76cef0..91b2737 100644
--- a/README.md
+++ b/README.md
@@ -4,9 +4,13 @@ This is Re-implementation of ShuffleNet, For details, please read the original p
 Neural Network for Mobile Devices" by Xiangyu Zhang et. al. 2017](https://arxiv.org/pdf/1707.01083.pdf). If you find any bug, please drop me an email. Thanks.
 
 # Notes  
-    | group 1  | group 2  |  group 3  |
-    | 1     2  | 3     4  |  5     6  |  
-    Each nubmer represents a channel of the feature map
+
+| group 1  | group 2  |  group 3  |  
+| :------: | :------: | :-------: |  
+| 1     2  | 3     4  |  5     6  |   
+
+Each nubmer represents a channel of the feature map  
+    
 ## step 1: Reshape  
 1  2  
 3  4   
@@ -15,8 +19,10 @@ Neural Network for Mobile Devices" by Xiangyu Zhang et. al. 2017](https://arxiv.
 1 3 5  
 2 4 6　　
 ## step 3: flatten  
-    | group 1  | group 2  |  group 3  |
-    | 1     3  | 5     2  |  4     6  |  
+
+| group 1  | group 2  |  group 3  |  
+| :-----:  | :------: | :-------: |  
+| 1     3  | 5     2  |  4     6  |  
 
 # Acknowledgement  
 [HolmesShuan](https://github.com/HolmesShuan)
diff --git a/shuffle_channel_layer.cpp b/shuffle_channel_layer.cpp
index 9351f6c..4b3b09a 100644
--- a/shuffle_channel_layer.cpp
+++ b/shuffle_channel_layer.cpp
@@ -11,6 +11,7 @@ void ShuffleChannelLayer<Dtype>::LayerSetUp(const vector<Blob<Dtype> *> &bottom,
     group_ = this->layer_param_.shuffle_channel_param().group();
     CHECK_GT(group_, 0) << "group must be greater than 0";
     temp_blob_.ReshapeLike(*bottom[0]);
+    top[0]->ReshapeLike(*bottom[0]);
 }
 
 template <typename Dtype>