Missing call to unittest.main()

Also made some PEP8 compliance adjustments