Python code for Udemy course - View it on GitHub