Create kalman_filter_4d.py