setup.py update

Signed-off-by: Kwpolska <kwpolska@gmail.com>