Think I made improvements in exercise3. Getting TypeError in test file?