Add tests :rooster: Probably will fail :sob: