I added tests and verified your current code works.