fixing tests 3,4. Still missing test 5.