updated for more regression