Cleans up the remove button a bit