Added a simple unittest