Minor changes
Added docs for ParamAugment
0.6.1