as you can see in the above image, I can only set one statement inside the function.

but I want to 

a.state = "xxx"

b.state = "xxx"

and I cannot select "custom"