Adding comments to problem solution
Fixing toString