Corrected test
Added another test