Assignment_2 complete