diff --git a/library/include/mobilisis/calculator_functions.h b/library/include/mobilisis/calculator_functions.h
index 3710c65..373885b 100644
--- a/library/include/mobilisis/calculator_functions.h
+++ b/library/include/mobilisis/calculator_functions.h
@@ -5,6 +5,7 @@
 #include <optional>
 
 #include "configuration.h"
+#include "calculate_price.h"
 #include "payment_method.h"
 #include "ticket.h"
 #include "tariff_time_range.h"
@@ -14,6 +15,7 @@ using namespace std;
 
 class Calculator {
     mutable QList<int> m_timeSteps;
+    mutable QList<int> m_priceSteps;
 
 protected:
     explicit Calculator() = default;
@@ -30,6 +32,9 @@ public:
     void ResetTimeSteps() { m_timeSteps.clear(); }
     QList<int> timeSteps() const { return m_timeSteps; }
 
+    void ResetPriceSteps() { m_priceSteps.clear(); }
+    QList<int> priceSteps() const { return m_priceSteps; }
+
 	/// <summary>
 	/// Gets duration in seconds from cost
 	/// </summary>