fixed correct answer was None problem