Initial code for quiz 3