FIX - was giving away the solution
Improved comments