Expanding on week 4 lab4.