Fixed missing update_vox_vectors and update_potential_energy function in snapshot_store.py