Add final solution code from Chris Lee @ Tealeaf Academy