Finished homework for week four