Sorry, I think I am being poor with my explanation.
My problem isn't / wasn't the execution of the code on the page, its a problem with the Data Assist Interface.
The OK button doesn't work when I select a WA rule. It only works when I select something like before page loads. So basically when I have a WA rule selected as a trigger I can't commit the changes I have made in the GUI to the code.
Its the fact that the OK button doesn't do anything in the behaviour window. It doesn't close it or commit my changes to the code.
If you need any more info I will do you a video to explain.