Adding shape inheritance [Python]