Added flag to keep branches