Add more to check answer function