updated repo + comments