Update rizki_test.py

Some more updates