Added check_stage script