edited and added functionality

Signed-off-by: Lendynyte <Dom1n1c@seznam.cz>