Minor changes adding stages