fixing up test
adding further tests