Adding comments to my solution