Homework 9 code..