Added Unit Test
Added additional test