Solved problem 4, I think..