Write more tests