Added model testing