added comment to prob 5 solution