Added comments + PEP 8 compliance
Combined with script/HestonSimulation.py