Use abs() function in a complex UFL form

I think its a similar issue to this:https://fenicsproject.discourse.group/t/nonlinear-complex-value-problem/13652
Will look into it a bit more to see if I can figure out how to approach it.