added some more stuff for regression