Update test.py

starting work on second gk function