Working on hw 4