Added comments to my solution