Added remove element function.