Cleaned up Python code