Changes to the vision code after the code review.