Added removeElement function