changed code comments a little