EE605 HW1 Code completed