Also forgot to add figures to repo