GRU assignment n_layers argument

Does this help?