Added Python code and a few notes