Minor documentation fix and renamed private class