- add couple more tests