Checkpoint
Added another unit test