New test
Fixed todo