Add missing specs, make code closer to diffeo demons plugin