This is a working version for test 5. However, Test1 fails