Finishing HW1 Demo