Modifications in code

Signed-off-by: Levi <l.lirvat@gmail.com>