Add my draft code for optimum_policy