Add status to IAI python class