WIP: test different optimization scheme