Adds some basic unit tests