Working on edit feature, currently works on one node