edited functionality

Signed-off-by: bikarevich <alexander.bikarevich@gmail.com>