Added remove function. Started working on edit feature. Added remove test.