problem 4 kinda working