Небольшие дополнения к коду

Signed-off-by: sadisasha <sadisasha@gmail.com>