Add root calculation
Refactoring test class