Day 2 challenge after stream code