add some comments after watching the 'official' solution