Extend cek.rb to match main script.