Add testing for later development of the gem