Few suggestion on the implementation

[omarqx]

Summery

Overall the code is pretty good, with few comments

Comments

• Make sure that the implementation works, when I ran the code it didn't work.
• Try to write task based commits into the code.
• Make sure to write the code into a pull request such as this.
• Some none coding specific practices such as indentation, comments, and naming conventions helps make the code more readable and easier to maintain.