Add tests for insertAfter
Implement insertAfter
Refactor remove method