Add figures and code to repo