working on problem 3, doesn't work yet