Fix for #13

Fixes #13 by adding roots to variables.