Changed main call to call correct class name (this must have been an error not noticed because __name__ == '__main__' was not called