Started work on figures for the paper