Add testing

[tremaineeto]

Need to add test code for the project.