Updated Controller to look at position instead of floor