diff --git a/webots/controllers/RL_Supervisor/rl_supervisor.py b/webots/controllers/RL_Supervisor/rl_supervisor.py
index 697abae9..309e9bd0 100644
--- a/webots/controllers/RL_Supervisor/rl_supervisor.py
+++ b/webots/controllers/RL_Supervisor/rl_supervisor.py
@@ -27,12 +27,22 @@
 # SOFTWARE.
 #
 
+
+################################################################################
+# Imports
+################################################################################
+
 import sys
 import struct
 from controller import Supervisor  # pylint: disable=import-error
 from serial_webots import SerialWebots
 from SerialMuxProt import Server
 
+################################################################################
+# Variables
+################################################################################
+
+
 # Constants
 ROBOT_NAME = "ROBOT"
 SUPERVISOR_RX_NAME = "serialComRx"
@@ -61,6 +71,11 @@
 ]
 
 
+################################################################################
+# Classes
+################################################################################
+
+
 class RobotController:
     """Class for data flow control logic"""
 
@@ -156,6 +171,12 @@ def process(self):
         self.__smp_server.process(self.__timestamp)
 
 
+################################################################################
+# Functions
+################################################################################
+
+
+# pylint: disable=duplicate-code
 def main_loop():
     """Main loop:
         - Perform simulation steps until Webots is stopping the controller.
@@ -235,3 +256,8 @@ def main_loop():
 
 
 sys.exit(main_loop())
+
+
+################################################################################
+# Main
+################################################################################