NW: fixing issues
working multi task learning, alpha version