Ruby version of the python problem I just added