Add tests :wrench: