Did this and wrote a test for it as well.