Update mlkit.rb

Implemented proposed changes