Fixed comments a bit