updated want button logic