Fixing issues with check_answer