Доработал решение, исходя из комментариев.