Moved __main__ into real test