Refresh UI after remove is pressed
Prevent wrong index via sorting