made some more changes

Signed-off-by: alyakhtar <samurai.aly@gmail.com>