Add testing

Add tests