Add status checking