After looking at Day 5 notes, decided to add a result variable as it seems easier to understand.