Updated vision code