homework 4th week wrote more code