Added unit testing: A few initial tests