write more tests