Find unit vector in a given direction (#605)