Complete up to and including remove function
Can't solve edit