Added ability to edit figures.