Add tests for feedforward