Add feed-forward term - to do