diff --git a/README.md b/README.md
index 487d249..220f2d9 100644
--- a/README.md
+++ b/README.md
@@ -1,7 +1,7 @@
 # Deep JSCC
 This implements training of deep JSCC models for wireless image transmission as described in the paper [Deep Joint Source-Channel Coding for Wireless Image Transmission](https://ieeexplore.ieee.org/abstract/document/8723589) by Pytorch. And there has been a [Tensorflow implementation](https://github.com/irdanish11/DJSCC-for-Wireless-Image-Transmission).
 
-This is my first time to use Pytorch and git to reproduce a paper, so there may be some mistakes. If you find any, please let me know. Thanks!
+This is my first time to use PyTorch and git to reproduce a paper, so there may be some mistakes. If you find any, please let me know. Thanks!
 
 ## Requirements
 
diff --git a/channel.py b/channel.py
new file mode 100644
index 0000000..e69de29
diff --git a/model.py b/model.py
new file mode 100644
index 0000000..e69de29
diff --git a/scripts.py b/scripts.py
new file mode 100644
index 0000000..e69de29