Optimalization for a small robot
Just a little change