Add more comments during check
Check if folder exists for setup check
Work on multi stages support
Fix no-stage case