Customize Customer Journey Plugin

I am trying to customize customer journey plugin code .

On click of complete checkbox on milestone task , it should display custom warning instead of completing milestone.

How can I customize "completeActivityClick"  function code so that it will be upgrade safe?