Changed a little with the self.net stuff

added self.net and removed net from argument in classify function