Got all the way to part two, stuck on remove function