added problem 4 starter code