have a working test, and working on second problem