Add velocity control