cleaned up jQuery, remove button now works