small update - class not finished