Added the first unit test