Add more tests :rooster: