Better documentation added. Now initial training comment possible. Fixed minor issues.