Expanding starter code for NW homework