Fix imports, missing variables in example