Adding both the code and the figures