Added a proper first unit test