TODO cleanup (#28)