added test. Still in import hell on the other branch.