cannot find an elegant way to put dropout
comparing with and without dropout