Energy recovery during braking