Как и обещал сделал вариант на основе референсов.