Added stochastic/conditional layers