Added in "trade after flat"