Add y/n check
Add y/n check