Removed unused import
Added testcase