Extra credit for Homework 2