Clean up of II.1.1-2.py

Added simple clean-up on data load.
Made result generating section more readable