Add draft code from behavioral model discussion