Просьба просто сделать то что попросили. В моих двух комментариях описаны несколько тестовых случаев, которые нужно добавить.