Add prompt and basic control flow