diff --git a/bbn_wave_freq_m5atomS3/KalmanForWave.h b/bbn_wave_freq_m5atomS3/KalmanForWave.h
index cbb499c..3d04a7e 100644
--- a/bbn_wave_freq_m5atomS3/KalmanForWave.h
+++ b/bbn_wave_freq_m5atomS3/KalmanForWave.h
@@ -6,8 +6,8 @@
 // create the filter structure
 #define KALMAN_NAME wave
 #define KALMAN_NUM_STATES 3
-#define KALMAN_NUM_INPUTS 0
-#include "FalmanFactoryFilter.h"
+#define KALMAN_NUM_INPUTS 1
+#include "KalmanFactoryFilter.h"
 
 // create the measurement structure
 #define KALMAN_MEASUREMENT_NAME displacement_integral
@@ -34,59 +34,58 @@ matrix_t *kalman_wave_get_state_transition(kalman_t *kf, matrix_data_t delta_t)
   return F;
 }
 
+matrix_t *kalman_wave_get_transition_offset(kalman_t *kf, matrix_data_t delta_t) {
+  matrix_t *B = kalman_get_input_transition(kf);
+  matrix_set(B, 0, 0, (matrix_data_t)(1.0 / 6.0) * delta_t * delta_t * delta_t);
+  matrix_set(B, 0, 1, (matrix_data_t)0.5 * delta_t * delta_t);
+  matrix_set(B, 0, 2, (matrix_data_t)delta_t);
+  return B;
+}
+
 void kalman_wave_init() {
 
   kalman_t *kf = kalman_filter_wave_init();
-  kalman_measurement_t *kfm = kalman_filter_wave_measurement_position_init();
+  kalman_measurement_t *kfm = kalman_filter_wave_measurement_displacement_integral_init();
 
+  // [KALMAN_NUM_STATES * 1]
   matrix_t *x = kalman_get_state_vector(kf);
   x->data[0] = 0.0; // displacement integral
   x->data[1] = 0.0; // vertical displacement
   x->data[2] = 0.0; // vertical velocity
 
-  // transition matrix
-  matrix_t *F = kalman_wave_get_state_transition(kf, delta_t);
+  // transition matrix [KALMAN_NUM_STATES * KALMAN_NUM_STATES]
+  //matrix_t *F = kalman_wave_get_state_transition(kf, delta_t);
 
-  // observation matrix
+  // observation matrix [KALMAN_NUM_MEASUREMENTS * KALMAN_NUM_STATES]
   matrix_t *H = kalman_get_measurement_transformation(kfm);
   matrix_set(H, 0, 0, (matrix_data_t)1.0);
   matrix_set(H, 0, 1, (matrix_data_t)0.0);
   matrix_set(H, 0, 2, (matrix_data_t)0.0);
 
-  // observation covariance
+  // observation covariance [KALMAN_NUM_MEASUREMENTS * KALMAN_NUM_MEASUREMENTS]
   matrix_t *R = kalman_get_process_noise(kfm);
   matrix_set(R, 0, 0, (matrix_data_t)1.0);
 
-  // transition offset
-  matrix_t *B = kalman_get_input_transition(kf);
-  matrix_set(B, 0, 0, (matrix_data_t)(1.0/6.0) * delta_t * delta_t * delta_t);
-  matrix_set(B, 0, 1, (matrix_data_t)0.5 * delta_t * delta_t);
-  matrix_set(B, 0, 2, (matrix_data_t)delta_t);
+  // transition offset [KALMAN_NUM_STATES * KALMAN_NUM_INPUTS]
   //kalman_wave_get_transition_offset(kf, delta_t);
 
-  // initial state covariance
+  // initial state covariance [KALMAN_NUM_STATES * KALMAN_NUM_STATES]
   matrix_t *P = kalman_get_system_covariance(kf);
-  matrix_set_symmetric(P, 0, 0, (matrix_data_t)1.0);  
-  matrix_set_symmetric(P, 0, 1, (matrix_data_t)0.0);  
-  matrix_set_symmetric(P, 0, 2, (matrix_data_t)0.0);  
-  matrix_set_symmetric(P, 1, 1, (matrix_data_t)1.0);  
-  matrix_set_symmetric(P, 1, 2, (matrix_data_t)0.0); 
+  matrix_set_symmetric(P, 0, 0, (matrix_data_t)1.0);
+  matrix_set_symmetric(P, 0, 1, (matrix_data_t)0.0);
+  matrix_set_symmetric(P, 0, 2, (matrix_data_t)0.0);
+  matrix_set_symmetric(P, 1, 1, (matrix_data_t)1.0);
+  matrix_set_symmetric(P, 1, 2, (matrix_data_t)0.0);
   matrix_set_symmetric(P, 2, 2, (matrix_data_t)1.0);
 
-  // transition covariance
+  // transition covariance [KALMAN_NUM_STATES * KALMAN_NUM_STATES]
   matrix_t *Q = kalman_get_process_noise(kf);
-  matrix_set_symmetric(Q, 0, 0, (matrix_data_t)1.0);  
-  matrix_set_symmetric(Q, 0, 1, (matrix_data_t)0.0);  
-  matrix_set_symmetric(Q, 0, 2, (matrix_data_t)0.0);  
-  matrix_set_symmetric(Q, 1, 1, (matrix_data_t)0.2);  
-  matrix_set_symmetric(Q, 1, 2, (matrix_data_t)0.0); 
+  matrix_set_symmetric(Q, 0, 0, (matrix_data_t)1.0);
+  matrix_set_symmetric(Q, 0, 1, (matrix_data_t)0.0);
+  matrix_set_symmetric(Q, 0, 2, (matrix_data_t)0.0);
+  matrix_set_symmetric(Q, 1, 1, (matrix_data_t)0.2);
+  matrix_set_symmetric(Q, 1, 2, (matrix_data_t)0.0);
   matrix_set_symmetric(Q, 2, 2, (matrix_data_t)0.1);
-     
-}
-
-{
-
-
 }
 
 #endif