Assignment 4 Python script