Updated docblocks
Updated to use ParamAugment
0.2.0