Update test.py

Hey i made cool changes!