Write calculate class, add sweet enum method