Update scheme.py

imporved