modified CNN module, it can work now!