Adding unit test

Just one