Edit functionality implimented