fixed bug probably
add more stages for exposition