Update test.py

Ég var að gera smá breytingar, sérðu þær?