Added extra credit for mission3