Adds some imgs
Tweaks and figure caption
Figure include working