Simplified Python script