work on a multi task learning w, seems to be much faster