Clean up Python a bit