added solution to remove-element