Added a pretty basic unit test, but a unit test nonetheless!