added couple more tests