EE605: HW5 solution added