Python POC: Added class diagram image