Added probabilitistic momentum model