Add some more figures to the code