Added test case, corrected formatting