Add more comments to solution