Adding first unit test