Fixed a bug where you click done and it doesn’t close out the edit screen