Added basic unit test