Check if answer is correct function is now working.