Added some additional comments to the solution