Added comments to solution