Done! I also added a unit test.