Plotting problems in separate directories, main.py refactored