Added some tests to address raised issues