- Corrected bug in solution