It looks like you're just implementing the CGC model in the PLE paper (or, of course, as a special case of PLE).