deleted test function by moving its functionality into main()
cleaned up formatting based on pylint