Initial code for HW0-2