Lagrangian force part done.. to do gradient still