Added extra credit stuff to mission 3