5. Added stages and basic menu interface