#2 - Added a test for the problem