Small design update

Close #3