Added rudimentary unit tests