updated the code to remove one element + CSS