Add full learning gain

Test Plan:
WIP