Clean up code

[SamimNiazi]

• Turn files into classes so be able to inherit certain methods and attributes. ✅
• Use a linter ✅
• Replace match-case in main.py by a function dictionary ❌ (It was easier just to displace match case from main)