Solution
Added test case