Sample code for HW1