Added answer check functionality