Edit functionality

--Added functionality for the edit box