You can remove the TODO since you've gone and done it :) but beyond that LGTM