Changed hw4 a bit