Add prompt and working branch selection