Done. Added a test case for it also.