Fixed formatting error and renamed test.py