trying to fix wrong answer