My ruby warrior code