Add unit test for removing a vertex