WIP: Add code to make a few figures