Went whitespace nazi.
Fixed branch implementation logic between stages 1 and 2