working code but doesn't actually do multi task learning, some more issues left to be fixed