Added first unit test