Last 2 TODOs
Removed stupid caption
Merge branch 'good_ol_plotting' of https://github.com/phimuemue/idp into good_ol_plotting