Added testing for remove(i) function