diff --git a/.github/workflows/pipeline.yml b/.github/workflows/pipeline.yml
index d8e3f06..eb0fc02 100644
--- a/.github/workflows/pipeline.yml
+++ b/.github/workflows/pipeline.yml
@@ -31,7 +31,7 @@ jobs:
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
       - name: Test files
-        run: pytest test/test_train.py
+        run: pytest src/test/test_train.py
 
   check_python_code_style:
     name: Check Python code style
@@ -74,7 +74,7 @@ jobs:
           if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
       - name: Train Model
-        run: python train.py
+        run: python src/train/train.py
 
   # build_image:
   #   name: Build image
diff --git a/app/main.py b/src/app/main.py
similarity index 100%
rename from app/main.py
rename to src/app/main.py
diff --git a/src/test/__init__.py b/src/test/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/test/test_train.py b/src/test/test_train.py
similarity index 96%
rename from test/test_train.py
rename to src/test/test_train.py
index 41b8b8a..b23e27e 100644
--- a/test/test_train.py
+++ b/src/test/test_train.py
@@ -2,7 +2,7 @@
 import pytest
 from tensorflow.keras.models import Sequential
 
-from train import read_data, create_model, train_model
+from ..train.train import read_data, create_model, train_model
 
 
 @pytest.fixture
diff --git a/src/train/__init__.py b/src/train/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/train.py b/src/train/train.py
similarity index 100%
rename from train.py
rename to src/train/train.py