Adds some unit tests