adding in more tests