Added code that passes all tests.