updated remove button but still not working