[#3] Added unit test