Finished edit function. need to change add and remove to thier own methods