Fixed bug in Main.py (creating regression models from group instead of category), refactored code in Main.py