Added Edit functionality