Model stages | Python