Found bug in the third case for remove but have not fixed it