Небольшое доковыривание кода

сабж