Ну человек спросил где ошибка, я написал. Дальше уже дело реализации, если человек разбирается как сделать самому то там можно и самому глянуть что выдает и где это в коде и как это переделать. Если решение нужно то обычно в комментах на learn js кидают кучу готовых решений. Типа смысл говорить как делать, если человек сам делать хочет?
спасибо за подсказку с return, разобрался (еще нужно было одну переменную добавить );