Add tool to extract state update logic from a proc.

Add tool to extract state update logic from a proc. #507