main.py file now plots the two period economy as well as the infinite horizon economy