Add link to main class & fix iteration comment