Writing more tests