WIP
fix updating correct answer