Added comments for tracking progress on training