Trying to separate out stages