Remove button works. UI is still creepy