Added editing functionality :D