updated ruby code