first draft of Python script