Reorganized and added a Python script