WhatsApp works to fix security flaw