Test Driven Development

Starting to add a few test cases into the project.