Lightning Challenge 2 solution