Added in first unit test