UCL Course RSD with Python - Assignment 4