Added class responsible for arguments in main.py file. Minor fixes.