1 files changed, 8 insertions, 5 deletions

diff --git a/tree.cpp b/tree.cpp
index ee45e57..340685f 100644
--- a/tree.cpp
+++ b/tree.cpp
@@ -1,21 +1,24 @@
 #include "tree.h"
-#include <math.h>
 
 Node::Node()
 {
 
 }
 
+template <class T> 
+double Node::castSolve (Node *node) {
+	T *tmp = static_cast<T*>( node );
+	return tmp->solve();
+}
+
 double Node::solve()
 {
 	switch (this->type) {
 		case OPERAND_NODE: {
-			OperandNode *tmp = static_cast<OperandNode*>( this );
-			return tmp->solve();
+			return this->castSolve<OperandNode>( this );
 		}
 		case OPERATOR_NODE: {
-			OperatorNode *tmp = static_cast<OperatorNode*>( this );
-			return tmp->solve();
+			return this->castSolve<OperatorNode>( this );
 		}
 	}
 }