Assignment 2 extra credit