Add testing stuff
Add test