Closes #48 by writing a simple unit test
Closes #49, simple tests passing
Removed sample code from the main project, will write some better code later