Add test :wrench: