refactor prep_nums function and update readme